package com.sw.c.utils;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.MalformedJwtException;
import io.jsonwebtoken.SignatureAlgorithm;
import io.jsonwebtoken.SignatureException;
import io.jsonwebtoken.UnsupportedJwtException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Function;

@Component
public class JwtTokenUtil {

    private static final Logger log = LoggerFactory.getLogger(JwtTokenUtil.class);
    // 从配置文件中注入 JWT 密钥
    @Value("${jwt.secret}")
    private String secret;

    // 从配置文件中注入 JWT 过期时间（秒）
    @Value("${jwt.expiration}")
    private Long expiration;

    // 从配置文件中注入 JWT 存储的请求头名称
    @Value("${jwt.header}")
    private String header;

    /**
     * 从 JWT 中提取用户名（subject）。
     *
     * @param token JWT 字符串
     * @return 用户名
     */
    public String getUsernameFromToken(String token) {
        try {
            return getClaimFromToken(token, Claims::getSubject);
        } catch (Exception e) {
            log.error("JWT 中提取用户名出现异常: {}", e.getMessage());
        }
            return null;
    }

    /**
     * 从 JWT 中提取过期时间。
     *
     * @param token JWT 字符串
     * @return 过期时间
     */
    public Date getExpirationDateFromToken(String token) {
        try {
            return getClaimFromToken(token, Claims::getExpiration);
        } catch (Exception e) {
            log.error("从 JWT 中提取过期时间出现异常: {}", e.getMessage());
        }
            return null;
    }

    /**
     * 根据提供的函数从 JWT 中提取特定的声明（claim）。
     *
     * @param token          JWT 字符串
     * @param claimsResolver 提取声明的函数
     * @param <T>            声明的类型
     * @return 特定的声明值
     */
    public <T> T getClaimFromToken(String token, Function<Claims, T> claimsResolver) {
        try {
            final Claims claims = getAllClaimsFromToken(token);
            return claimsResolver.apply(claims);
        } catch (Exception e) {
            log.error("从 JWT 中提取过期时间出现异常: {}", e.getMessage());
        }
            return null;
    }

    /**
     * 解析 JWT 并返回所有的声明。
     *
     * @param token JWT 字符串
     * @return 所有的声明
     */
    private Claims getAllClaimsFromToken(String token) {
        try {
            return Jwts.parser().setSigningKey(secret).parseClaimsJws(token).getBody();
        } catch (Exception e) {
            log.error("解析 JWT出现异常: {}", e.getMessage());
        }
        return null;
    }

    /**
     * 检查 JWT 是否已经过期。
     *
     * @param token JWT 字符串
     * @return 如果过期则返回 true，否则返回 false
     */
    private Boolean isTokenExpired(String token) {
        try {
            final Date expiration = getExpirationDateFromToken(token);
            return expiration.before(new Date());
        } catch (Exception e) {
            log.error("Error checking if token is expired: {}", e.getMessage());
            return true; // 默认认为过期以提高安全性
        }
    }

    /**
     * 根据用户名生成一个新的 JWT。
     *
     * @param username 用户名
     * @return 新的 JWT 字符串
     */
    public String generateToken(String username) {
        Map<String, Object> claims = new HashMap<>();
        return doGenerateToken(claims, username);
    }

    /**
     * 使用提供的声明和主题生成 JWT。
     *
     * @param claims  自定义声明
     * @param subject 主题（通常是用户名）
     * @return 生成的 JWT 字符串
     */
    private String doGenerateToken(Map<String, Object> claims, String subject) {
        try {
            return Jwts.builder()
                    .setClaims(claims)
                    .setSubject(subject)
                    .setIssuedAt(new Date(System.currentTimeMillis()))
                    .setExpiration(new Date(System.currentTimeMillis() + expiration * 1000))
                    .signWith(SignatureAlgorithm.HS512, secret)
                    .compact();
        } catch (Exception e) {
            log.error("Error generating JWT token: {}", e.getMessage());
            return null;
        }
    }

    /**
     * 验证 JWT 是否有效，即用户名是否匹配且未过期。
     *
     * @param token         JWT 字符串
     * @param userDetails 用户详情
     * @return 如果有效则返回 true，否则返回 false
     */
    public Boolean validateToken(String token, UserDetails userDetails) {
        try {
            final String username = getUsernameFromToken(token);
            return (username.equals(userDetails.getUsername()) && !isTokenExpired(token));
        } catch (Exception e) {
            log.error("Error validating JWT token: {}", e.getMessage());
            return false;
        }
    }

    /**
     * 返回存储 JWT 的 HTTP 请求头名称。
     *
     * @return 请求头名称
     */
    public String getHeader() {
        return header;
    }
}



